package com.ruoyi.system.api.generator;

import com.baomidou.mybatisplus.enums.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.rules.DbType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

import java.sql.SQLException;

/**
 * @ClassName: MyBatisPlusGenerator
 * @Description: 代码生成工具
 * @Author: LZH
 * @Date: 2020/7/21 0:53
 * @Version: 1.0
 */
public class MyBatisPlusGenerator {

    public static void main(String[] args) throws SQLException {

        /** 1. 全局配置 **/
        GlobalConfig config = new GlobalConfig();
        // 是否支持AR模式
        config.setActiveRecord(true)
                // 作者
                .setAuthor("LZH")
                // 生成路径
                .setOutputDir("D:\\project\\重庆沙坪坝枢纽站系统\\RuoYi-Cloud\\ruoyi-file\\src\\main\\java")
                // 文件覆盖
                .setFileOverride(true)
                // 主键策略
                .setIdType(IdType.AUTO)
                // 设置生成的service接口的名字的首字母是否为I
                .setServiceName("%sService")
                // 生成基本的resultMap
                .setBaseResultMap(true)
                // 生成基本的SQL片段
                .setBaseColumnList(true);

        /** 2. 数据源配置 **/
        DataSourceConfig dsConfig  = new DataSourceConfig();
        // 设置数据库类型
        dsConfig.setDbType(DbType.MYSQL)
                .setDriverName("com.mysql.cj.jdbc.Driver")
                .setUrl("jdbc:mysql://192.168.2.36:3306/ry-cloud")
                .setUsername("root")
                .setPassword("123456");

        /** 3. 策略配置globalConfiguration中 **/
        StrategyConfig stConfig = new StrategyConfig();
        // 全局大写命名
        stConfig.setCapitalMode(true)
                // 指定表名 字段名是否使用下划线
                // .setDbColumnUnderline(true)
                // 数据库表映射到实体的命名策略
                .setNaming(NamingStrategy.underline_to_camel)
                // 生成的表
                .setInclude("file_record");

        /** 4. 包名策略配置 **/
        PackageConfig pkConfig = new PackageConfig();
        pkConfig.setParent("com.ruoyi.file")
                .setMapper("dao")
                .setService("service")
                .setController("controller")
                .setEntity("model")
                .setXml("mapper");

        /** 5. 整合配置 **/
        AutoGenerator ag = new AutoGenerator();
        ag.setGlobalConfig(config)
                .setDataSource(dsConfig)
                .setStrategy(stConfig)
                .setPackageInfo(pkConfig);

        /** 6. 执行 **/
        ag.execute();
    }
}
